﻿
// ToolDlg.cpp: 实现文件
//

#include "pch.h"
#include "framework.h"
#include "Tool.h"
#include "ToolDlg.h"
#include "afxdialogex.h"

#include <exception>
#include "lua.hpp"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif

#define RT_OK	(0)
#define RT_FAIL	(-1)

// 全局指针，用于访问对话框类的实例
CToolDlg* g_pDialog = nullptr;


// Catch C++ exceptions and convert them to Lua error messages.
// Customize as needed for your own exception classes.
static int wrap_exceptions(lua_State* L, lua_CFunction f)
{
	try {
		return f(L);  // Call wrapped function and return result.
	}
	catch (const char* s) {  // Catch and convert exceptions.
		lua_pushstring(L, s);
	}
	catch (std::exception& e) {
		lua_pushstring(L, e.what());
	}
	catch (...) {
		lua_pushliteral(L, "caught (...)");
	}
	return lua_error(L);  // Rethrow as a Lua error.
}

static int myinit(lua_State* L)
{
	// Define wrapper function and enable it.
	lua_pushlightuserdata(L, (void*)wrap_exceptions);
	luaJIT_setmode(L, -1, LUAJIT_MODE_WRAPCFUNC | LUAJIT_MODE_ON);
	lua_pop(L, 1);

	return 0;
}



// 自定义的 C 函数，用于替代 print
static int myPrint(lua_State* L) {
	CString output;  // 用于存储输出的字符串

	int n = lua_gettop(L);  // 获取栈上参数的数量

	for (int i = 1; i <= n; i++) {
		const char* str = lua_tostring(L, i);  // 获取参数的字符串表示
		if (str) {
			output += CString(str);
		}
		else {
			output += _T("[non-string]");
		}

		if (i < n) {
			output += _T("\t");
		}
	}

	output += CString("\r\n");

	// 将输出追加到编辑框
	if (g_pDialog) {
		g_pDialog->setEditMessage(output);
	}

	return 0;  // 返回值表示压入栈的结果数量
}

// 用于应用程序“关于”菜单项的 CAboutDlg 对话框

class CAboutDlg : public CDialogEx
{
public:
	CAboutDlg();

// 对话框数据
#ifdef AFX_DESIGN_TIME
	enum { IDD = IDD_ABOUTBOX };
#endif

	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持

// 实现
protected:
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialogEx(IDD_ABOUTBOX)
{
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialogEx::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx)
END_MESSAGE_MAP()


// CToolDlg 对话框



CToolDlg::CToolDlg(CWnd* pParent /*=nullptr*/)
	: CDialogEx(IDD_TOOL_DIALOG, pParent)
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CToolDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialogEx::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CToolDlg, CDialogEx)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_BTN_RELOAD, &CToolDlg::OnClickedBtnReload)
END_MESSAGE_MAP()


// CToolDlg 消息处理程序

BOOL CToolDlg::OnInitDialog()
{
	CDialogEx::OnInitDialog();

	// 将“关于...”菜单项添加到系统菜单中。

	// IDM_ABOUTBOX 必须在系统命令范围内。
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != nullptr)
	{
		BOOL bNameValid;
		CString strAboutMenu;
		bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX);
		ASSERT(bNameValid);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

	// 设置此对话框的图标。  当应用程序主窗口不是对话框时，框架将自动
	//  执行此操作
	SetIcon(m_hIcon, TRUE);			// 设置大图标
	SetIcon(m_hIcon, FALSE);		// 设置小图标

	// TODO: 在此添加额外的初始化代码
	g_pDialog = this;
	// 在初始化代码中添加以下行
	//freopen("CONOUT$", "w", stdout);

	return TRUE;  // 除非将焦点设置到控件，否则返回 TRUE
}

void CToolDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialogEx::OnSysCommand(nID, lParam);
	}
}

// 如果向对话框添加最小化按钮，则需要下面的代码
//  来绘制该图标。  对于使用文档/视图模型的 MFC 应用程序，
//  这将由框架自动完成。

void CToolDlg::OnPaint()
{
	if (IsIconic())
	{
		CPaintDC dc(this); // 用于绘制的设备上下文

		SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

		// 使图标在工作区矩形中居中
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// 绘制图标
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialogEx::OnPaint();
	}
}

//当用户拖动最小化窗口时系统调用此函数取得光标
//显示。
HCURSOR CToolDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}

int CToolDlg::setEditMessage(CString message) 
{
	// 获取编辑框控件指针
	CEdit* pEdit = (CEdit*)GetDlgItem(IDC_EDIT_DEBUG);

	// 设置编辑框文本
	if (pEdit) {
		CString currentText;
		pEdit->GetWindowText(currentText);
		currentText += message;
		pEdit->SetWindowText(currentText);
	}
	return RT_OK;
}


void CToolDlg::OnClickedBtnReload()
{
	CString str = this->getScriptFilePath();
	// 获取可写入的 LPTSTR 指针
	LPTSTR buffer = str.GetBuffer();

	// 将 LPTSTR 转换为 char*
	const char* scriptFile = "test.lua";// static_cast<const char*>(CT2A(buffer));

	lua_State* L;
	// TODO: 在此添加控件通知处理程序代码
	TRACE("clicke reload button");
	L = lua_open();
	luaL_openlibs(L); // 打开Lua标准库

	//myinit(L);

	// 注册自定义的 C 函数 myPrint
	lua_pushcfunction(L, myPrint);
	lua_setglobal(L, "DebugPrint");

	if (luaL_loadfile(L, scriptFile) || lua_pcall(L, 0, 0, 0)) {
		// 处理加载和运行错误
		const char* szMsg = lua_tostring(L, -1);
		AfxMessageBox(_T("脚本错误：") + CString(szMsg));
	}
	lua_close(L);

	// 释放缓冲区
	str.ReleaseBuffer();
}

CString CToolDlg::getScriptFilePath() 
{
	if (this->isCustomFilePath()) {
		return this->getOpenFilePath();
	}

	return CString("test.lua");
}

CString CToolDlg::getOpenFilePath()
{
	// 声明一个CFileDialog对象
	CFileDialog fileDlg(TRUE, NULL, NULL, OFN_FILEMUSTEXIST | OFN_HIDEREADONLY, _T("All Files (*.*)|*.*||"), AfxGetMainWnd());

	// 显示文件对话框
	if (fileDlg.DoModal() == IDOK)
	{
		// 获取所选文件的路径
		CString filePath = fileDlg.GetPathName();

		// 在这里，您可以使用filePath进行后续操作
		// 例如，打开文件、读取文件内容等

		// 示例：打开文件
		//CFile file;
		//if (file.Open(filePath, CFile::modeRead))
		//{
		//	// 在这里进行文件操作，如读取文件内容
		//	file.Close();
		//}
		return filePath;
	}

	return CString("");
}

BOOL CToolDlg::isCustomFilePath()
{
	TRACE(_T("是否需要打开"));
	CButton* pCheckBox = (CButton*)GetDlgItem(IDC_CHECK_CUSTOM_FILE_PATH);

	// 获取复选框的状态
	int checkBoxState = pCheckBox->GetCheck();

	// 判断复选框是否被选中
	if (checkBoxState == BST_CHECKED)
	{
		// 复选框被选中
		return TRUE;
	}
	else if (checkBoxState == BST_UNCHECKED)
	{
		// 复选框未被选中
		return FALSE;
	}

	return FALSE;
}